# Please install OpenAI SDK first: `pip3 install openai`
"""
出于与 OpenAI 兼容考虑，您也可以将 base_url 设置为 https://api.deepseek.com/v1 来使用，但注意，此处 v1 与模型版本无关。
* deepseek-chat 模型已全面升级为 DeepSeek-V3，接口不变。 通过指定 model='deepseek-chat' 即可调用 DeepSeek-V3。
* deepseek-reasoner 是 DeepSeek 最新推出的推理模型 DeepSeek-R1。通过指定 model='deepseek-reasoner'，即可调用 DeepSeek-R1。
"""
from openai import OpenAI
import openai

client = OpenAI(api_key="sk-ba0f81b0e34a4f029aba0ef2cb8c22f1", base_url="https://api.deepseek.com")

try:
    response = client.chat.completions.create(
        model="deepseek-chat",
        messages=[
            {"role": "system", "content": "You are a helpful assistant"},
            {"role": "user", "content": "Hello"},
        ],
        stream=False
    )
except openai.APIStatusError as e:
    if "Insufficient Balance" in str(e):
        print("错误：账户余额不足，请充值")
    else:
        print(f"API错误: {e}")
else:
    print(response.choices[0].message.content)